package com.wms.handlerinterceptor;

import com.wms.context.BaseContext;
import com.wms.entity.WmsUser;
import com.wms.vo.UserRoleVo;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import org.springframework.web.method.HandlerMethod;
import org.springframework.web.servlet.HandlerInterceptor;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

@Component
@Slf4j
public class BaseContextInterceptor implements HandlerInterceptor {
    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
        BaseContext.removeCurrentId();
    }

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        //判断当前拦截到的是Controller的方法还是其他资源
        if (!(handler instanceof HandlerMethod)) {
            //当前拦截到的不是动态方法，直接放行
            return true;
        }
        UserRoleVo user = (UserRoleVo)request.getSession().getAttribute("user");
        if (user != null) {
            BaseContext.setCurrentId(user.getUserId());
        }
        log.info("当前用户id为：{}", BaseContext.getCurrentId());
        return true;
    }
}
